PyTorch 0.4.0 大更新,正式支持 Windows 平台
(点击上方公众号,可快速关注)
参考:PyTorch、开源中国,
整理:Python开发者(id:PythonCoder)
PyTorch 是一个 Python 优先的深度学习框架,提供两个高级功能:
强大的 GPU 加速 Tensor 计算(类似 numpy)
构建基于 tape 的自动升级系统上的深度神经网络
4 月 25 日,PyTorch 的 GitHub 主页宣布 PyTorch 0.4.0 已发布。
此次新版,除了平衡计算内存、支持更多概率分布、优化性能和修复 Bug 外,PyTorch 还正式官方支持 Windows 系统,不再需要借助其它开发者发布的第三方 conda 包。为 Python 3.5 和 3.6 提供预编译的 Conda 二进制文件和 pip wheels。
另外,Windows 上的 PyTorch 不支持分布式训练,可能比 Linux/OSX 慢一点,因为 Visual Studio 支持较早版本的 OpenMP。
Windows 用户遇到问题,请看这个 FAQ:
http://pytorch.org/docs/stable/notes/windows.html
PyTorch更新列表:
主要核心更改
Tensor / Variable 合并
Zero-dimensional Tensors
数据类型
迁移指南 (http://pytorch.org/2018/04/22/0_4_0-migration-guide.html)
新特性
支持 RNN
易于使用的 Launcher utility
NCCL2 后端
24 个基础的概率分布
新增 cdf, variance, entropy, perplexity 等
平衡计算内存
bottleneck - 一款识别代码中的热点的工具
全面支持高级索引
快速傅立叶转换
Tensors
Neural Networks
torch.distributions
Distributed Training
C++ 拓展
支持 Windows
ONNX Improvements
具体细节,请查看 PyTorch 官方发行说明:
https://github.com/pytorch/pytorch/releases/tag/v0.4.0
看完本文有收获?请转发分享给更多人
关注「Python开发者」,提升Python技能